/*
 * Copyright 2024 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.pajamas.starter.integration.module.client;

import org.apache.commons.collections4.MapUtils;
import org.apache.commons.lang3.ArrayUtils;
import org.apache.commons.lang3.StringUtils;
import org.pajamas.starter.core.util.RegisterAsPajamasClient;
import org.pajamas.starter.integration.spring.bean.AbstractSimpleBeanDefRegistrar;
import org.pajamas.starter.integration.spring.bean.CandidateBeanDefSetBuilder;
import org.springframework.core.env.Environment;
import org.springframework.core.io.ResourceLoader;
import org.springframework.core.type.AnnotationMetadata;

import java.util.Arrays;
import java.util.Map;

/**
 * @author william
 * @since 2024/2/5
 */
public class PajamasClientConfBeanRegistrar extends AbstractSimpleBeanDefRegistrar {
    private static final String KEY_BASE_PACKAGES_CUSTOMIZE;

    static {
        KEY_BASE_PACKAGES_CUSTOMIZE = "basePackages";
    }

    protected PajamasClientConfBeanRegistrar(Environment environment, ResourceLoader resourceLoader) {
        super(environment, resourceLoader);
    }

    @Override
    protected void doRegisterBeanDefinitions(AnnotationMetadata importingClassMetadata,
                                             CandidateBeanDefSetBuilder beanDefSetBuilder) {
        beanDefSetBuilder
                .currentBasePackage()
                .starterBasePackage()
                .annotated(RegisterAsPajamasClient.ConfigurationBean.class);

        Map<String, Object> annAttr =
                importingClassMetadata.getAnnotationAttributes(RegisterAsPajamasClient.class.getName());
        if (MapUtils.isNotEmpty(annAttr) && annAttr.containsKey(KEY_BASE_PACKAGES_CUSTOMIZE)) {
            String[] basePackages = (String[]) annAttr.get(KEY_BASE_PACKAGES_CUSTOMIZE);
            if (ArrayUtils.isNotEmpty(basePackages)) {
                Arrays.stream(basePackages)
                        .filter(StringUtils::isNotBlank)
                        .map(String::trim)
                        .distinct()
                        .forEach(beanDefSetBuilder::basePackage);
            }
        }
    }
}
